from functools import wraps
from flask import g, redirect, url_for


def login_required(func):
    # 保留func信息
    @wraps(func)
    def inner(*args, **kwargs):
        # 判断用户是否登录
        if g.user:
            # 用户已登录，则执行原函数
            return func(*args, **kwargs)
        else:
            # 用户未登录，则重定向到登录页面
            return redirect(url_for('User_Auth.login'))

    return inner
